import onnxruntime as ort

class ONNXRuntime:
    def __init__(self, model_path):
        self.session = ort.InferenceSession(model_path)

    def run(self, inputs):
        return self.session.run(None, inputs)